一 . 概述

它能将一个含任意实数的K维向量 z “压缩”到另一个K维实向量 中,使得每一个元素的范围都在(0, 1)之间,并且所有元素的和为1(也可视为一个 (k-1)维的hyperplan,因为总和为1,所以是subspace)。

二 . 实现代码

1
2
3
4
5
6
7
8
9
import numpy as np

l = np.array([-1, 0, 1])

def softmax(l):
exp = np.exp(l)
return np.divide(exp, exp.sum())

softmax(l)